# pip install zhipuai 请先在终端进行安装
import os

from zhipuai import ZhipuAI

client = ZhipuAI(api_key=os.environ["ZHIPU_API_KEY_1"])

#glm-4大模型现在通过提示，是无法一次性写对有难度的代码的

while True:

    # prompt = "请写一段代码。现在我想实现一个基于AlexNet的网络架构\
    # 这个网络架构有两部分输入，这两部分输入结构不一致，但在架构的中间两部分输入的结构会合并\
    # 请你构建这个架构，并假设两个输入数据，验证这个网络能够顺利运行"

    prompt = input("请输入您的问题？")
    if prompt == 'exit':
        break

    response = client.chat.completions.create(
        model="glm-4-flash",  # 填写需要调用的模型名称
        messages=[
            {"role": "user", "content": prompt}
        ],
        stream=True
    )

    for chunk in response:
        print(chunk.choices[0].delta.content, end='')
    print()
    # print(response.choices[0].message.content)
